/*================================================================
*   Copyright (C) 2017 LiuYao. All rights reserved.
*   
*   Filename : transpose_op.h
*   Author : LiuYao
*   Date : 2017-12-22
*   Description : 
*
================================================================*/
#ifndef _TRANSPOSE_OP_H
#define _TRANSPOSE_OP_H

#include <vector>
#include "../node.h"
#include "op.h"
#include "../../session/context.h"

using std::vector;

class Transpose : public OpNode {
    public:
        Transpose(Node* );
        vector<Node*> getGradNodes(Node* );
        void compute(Context* );
};

#endif
